home *** CD-ROM | disk | FTP | other *** search
/ PC World 2007 March / PCWorld_2007-03_cd.bin / v cisle / httrack / httrack-3.41-2.exe / {app} / src / proxy / main.c < prev    next >
Encoding:
C/C++ Source or Header  |  2006-10-15  |  5.5 KB  |  184 lines
  1. /* ------------------------------------------------------------ */
  2. /*
  3. HTTrack Website Copier, Offline Browser for Windows and Unix
  4. Copyright (C) Xavier Roche and other contributors
  5.  
  6. This program is free software; you can redistribute it and/or
  7. modify it under the terms of the GNU General Public License
  8. as published by the Free Software Foundation; either version 2
  9. of the License, or any later version.
  10.  
  11. This program is distributed in the hope that it will be useful,
  12. but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14. GNU General Public License for more details.
  15.  
  16. You should have received a copy of the GNU General Public License
  17. along with this program; if not, write to the Free Software
  18. Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19.  
  20. Please visit our Website: http://www.httrack.com
  21. */
  22.  
  23. /* ------------------------------------------------------------ */
  24. /* File: ProxyTrack, httrack cache-based proxy                  */
  25. /* Author: Xavier Roche                                         */
  26. /* ------------------------------------------------------------ */
  27.  
  28. /* Standard includes */
  29. #include <stdio.h>
  30. #include <stdlib.h>
  31. #include <string.h>
  32. #include <time.h>
  33. #include <ctype.h>
  34.  
  35. #include "htsbase.h"
  36. #include "htsnet.h"
  37. #include "htslib.h"
  38. #include "store.h"
  39. #include "proxytrack.h"
  40.  
  41. #ifndef _WIN32
  42. #include <signal.h>
  43. static void sig_brpipe( int code ) {
  44.   /* ignore */
  45. }
  46. #endif
  47.  
  48. static int scanHostPort(const char* str, char *host, int *port) {
  49.     char* pos = strrchr(str, ':');
  50.     if (pos != NULL) {
  51.         int n = (int) ( pos - str );
  52.         if (n < 256) {
  53.             host[0] = '\0';
  54.             strncat(host, str, n);
  55.             if (sscanf(pos + 1, "%d", port) == 1) {
  56.                 return 1;
  57.             }
  58.         }
  59.     }
  60.     return 0;
  61. }
  62.  
  63. int main(int argc, char* argv[])
  64. {
  65.   int i;
  66.   int ret = 0;
  67.     int proxyPort = 0, icpPort = 0;
  68.     char proxyAddr[256 + 1], icpAddr[256 + 1];
  69.     PT_Indexes index;
  70.  
  71. #ifdef _WIN32
  72.   {
  73.     WORD   wVersionRequested;   // requested version WinSock API
  74.     WSADATA wsadata;            // Windows Sockets API data
  75.     int stat;
  76.     wVersionRequested = 0x0101;
  77.     stat = WSAStartup( wVersionRequested, &wsadata );
  78.     if (stat != 0) {
  79.       fprintf(stderr, "Winsock not found!\n");
  80.       return -1;
  81.     } else if (LOBYTE(wsadata.wVersion) != 1  && HIBYTE(wsadata.wVersion) != 1) {
  82.       fprintf(stderr, "WINSOCK.DLL does not support version 1.1\n");
  83.       WSACleanup();
  84.       return -1;
  85.     }
  86.   }
  87. #endif
  88.  
  89.     /* Args */
  90.     printf("ProxyTrack %s, build proxies upon HTTrack Website Copier Archives\n", PROXYTRACK_VERSION);
  91.     printf("Copyright (C) Xavier Roche and other contributors\n");
  92.     printf("\n");
  93.     printf("This program is free software; you can redistribute it and/or\n");
  94.     printf("modify it under the terms of the GNU General Public License\n");
  95.     printf("as published by the Free Software Foundation; either version 2\n");
  96.     printf("of the License, or any later version.\n");
  97.     printf("\n");
  98.     printf("*** This version is a development release ***\n");
  99.     printf("\n");
  100.     if (argc < 3
  101.         || (
  102.             strcmp(argv[1], "--convert") != 0
  103.             &&
  104.             (
  105.                 !scanHostPort(argv[1], proxyAddr, &proxyPort)
  106.                 || !scanHostPort(argv[2], icpAddr, &icpPort)
  107.             )
  108.             )
  109.         ) 
  110.     {
  111.         fprintf(stderr, "proxy mode:\n");
  112.         fprintf(stderr, "usage: %s <proxy-addr:proxy-port> <ICP-addr:ICP-port> [ ( <new.zip path> | <new.ndx path> | <archive.arc path> | --list <file-list> ) ..]\n", argv[0]);
  113.         fprintf(stderr, "\texample:%s proxy:8080 localhost:3130 /home/archives/www-archive-01.zip /home/old-archives/www-archive-02.ndx\n", argv[0]);
  114.         fprintf(stderr, "convert mode:\n");
  115.         fprintf(stderr, "usage: %s --convert <archive-output-path> [ ( <new.zip path> | <new.ndx path> | <archive.arc path> | --list <file-list> ) ..]\n", argv[0]);
  116.         fprintf(stderr, "\texample:%s proxy:8080 localhost:3130 /home/archives/www-archive-01.zip /home/old-archives/www-archive-02.ndx\n", argv[0]);
  117.     return 1;
  118.   }
  119.     index = PT_New();
  120.     for(i = 3 ; i < argc ; i++) {
  121.         if (argv[i][0] == '-') {
  122.             if (strcmp(argv[i], "--list") == 0) {
  123.                 if (i + 1 < argc) {
  124.                     char line[256 + 1];
  125.                     FILE *fp = fopen(argv[++i], "rb");
  126.                     if (fp == NULL) {
  127.                         fprintf(stderr, "error: could not process list %s\n", argv[i]);
  128.                         exit(1);
  129.                     }
  130.                     while(linput(fp, line, 256)) {
  131.                         int itemsAdded = PT_AddIndex(index, line);
  132.                         if (itemsAdded > 0) {
  133.                             fprintf(stderr, "processed: %s (%d items added)\n", line, itemsAdded);
  134.                         } else if (itemsAdded == 0) {
  135.                             fprintf(stderr, "processed: %s (no items added)\n", line);
  136.                         } else {
  137.                             fprintf(stderr, "error: could not process %s\n", line);
  138.                         }
  139.                     }
  140.                     fclose(fp);
  141.                 }
  142.             } else {
  143.                 fprintf(stderr, "* bad arg %s\n", argv[i]);
  144.                 exit(1);
  145.             }
  146.         } else {
  147.             int itemsAdded = PT_AddIndex(index, argv[i]);
  148.             if (itemsAdded > 0) {
  149.                 fprintf(stderr, "processed: %s (%d items added)\n", argv[i], itemsAdded);
  150.             } else if (itemsAdded == 0) {
  151.                 fprintf(stderr, "processed: %s (no items added)\n", argv[i]);
  152.             } else {
  153.                 fprintf(stderr, "error: could not process %s\n", argv[i]);
  154.             }
  155.         }
  156.     }
  157.  
  158.     /* sigpipe */
  159. #ifndef _WIN32
  160.   signal( SIGPIPE , sig_brpipe );   // broken pipe (write into non-opened socket)
  161. #endif
  162.  
  163.   /* Go */
  164.     if (strcmp(argv[1], "--convert") != 0) {
  165.         ret = proxytrack_main(proxyAddr, proxyPort, icpAddr, icpPort, index);
  166.     } else {
  167.         if ((ret = PT_SaveCache(index, argv[2])) == 0) {
  168.             fprintf(stderr, "processed: '%s'\n", argv[2]);
  169.         } else {
  170.             fprintf(stderr, "error: could not save '%s'\n", argv[2]);
  171.         }
  172.     }
  173.  
  174.     /* Wipe */
  175.     PT_Delete(index);
  176.  
  177. #ifdef _WIN32
  178.   WSACleanup();
  179. #endif
  180.  
  181.   return ret;
  182. }
  183.  
  184.